import glob
import os
import concurrent.futures
from datetime import datetime, timedelta
from login import login
from 获取订单 import request_get, get_order_list
from page import get_page
from 获取聊天数据 import get_message_data
from 回调 import response_data
from rich.progress import track

# 账号
usernames1 = {'username': 'yexiong1997@qq.com', 'password': 'Yexiong1997', 'port': 9225, 'user_path': r"C:\15867710780",
              'down_path': r"C:\15867710780\filedownloads"}
usernames2 = {'username': '594688317@qq.com', 'password': 'Yh5566789', 'port': 9223, 'user_path': r"C:\594688317",
              'down_path': r"C:\594688317\filedownloads"}
usernames3 = {'username': '791002765@qq.com', 'password': 'yexiong1997Mm', 'port': 9224, 'user_path': r"C:\791002765",
              'down_path': r"C:\791002765\filedownloads"}
usernames4 = {'username': '2968425169@qq.com', 'password': 'yexiong1997Mm', 'port': 9226, 'user_path': r"C:\2968425169",
              'down_path': r"C:\2968425169\filedownloads"}
usernames5 = {'username': 'yexiong1108@qq.com', 'password': 'yexiong1997Mm', 'port': 9240,
              'user_path': r"C:\yexiong1108", 'down_path': r"C:\yexiong1108\filedownloads"}
usernames6 = {'username': 'y2968425169@foxmail.com', 'password': 'Yexiong1997', 'port': 9241,
              'user_path': r"C:\y2968425169", 'down_path': r"C:\y2968425169\filedownloads"}
shop_dirt = {'抖音小店-昀财好物严选': 'yexiong1997@qq.com', '抖音小店-抖in优选供应链': 'yexiong1997@qq.com',
             '抖音小店-埠屯百货商行': 'yexiong1997@qq.com', '抖音小店-巨晶家居旗舰店': 'yexiong1997@qq.com',
             '抖音小店-首力五金工具旗舰店': 'yexiong1997@qq.com', '抖音小店-首力建材旗舰店': 'yexiong1997@qq.com',
             '抖音小店-昀恒好物严选': 'yexiong1997@qq.com', '抖音小店-淘淘家穿戴甲': 'yexiong1997@qq.com',
             '抖音小店-首捷建材商行': 'yexiong1997@qq.com', '抖音小店-首力家装主材旗舰店': 'yexiong1997@qq.com',
             '抖音小店-柔乾建材商行': 'yexiong1997@qq.com', '抖音小店-埠芳建材商行': 'yexiong1997@qq.com',
             '抖音小店-治野建材商行': 'yexiong1997@qq.com', '抖音小店-逅恒百货商行': 'yexiong1997@qq.com',
             '抖音小店-韵晗百货商行': 'yexiong1997@qq.com', '抖音小店-首力智能家装旗舰店': 'yexiong1997@qq.com',
             '抖音小店-昀铎建材商行': 'yexiong1997@qq.com', '抖音小店-月容专营店': 'yexiong1997@qq.com',
             '抖音小店-昀宏建材商行': 'yexiong1997@qq.com', '抖音小店-豪有钱好物优选': 'yexiong1997@qq.com',
             '抖音小店-巨晶昀沐家居日用专卖店': 'yexiong1997@qq.com', '抖音小店-巨晶旗舰店': 'yexiong1997@qq.com',
             '抖音小店-忆菡好物优选': '594688317@qq.com', '抖音小店-俄璐专营店': '594688317@qq.com',
             '抖音小店-真子家居日用旗舰店': '594688317@qq.com', '抖音小店-朋意官方旗舰店': '594688317@qq.com',
             '抖音小店-亿居百货': '594688317@qq.com', '抖音小店-首力建材商行': '594688317@qq.com',
             '抖音小店-昀川供应链': '594688317@qq.com',
             '抖音小店-朋意义乌市海辉服饰有限公司烹饪用具专卖店': '594688317@qq.com',
             '抖音小店-昀初百货': '594688317@qq.com', '抖音小店-沐果家居专营店': '594688317@qq.com',
             '抖音小店-首力官方旗舰店': '594688317@qq.com', '抖音小店-悦琦好物优选': '594688317@qq.com',
             '抖音小店-朋意烹饪用具旗舰店': '594688317@qq.com', '抖音小店-欣茉好物优选': '594688317@qq.com',
             '抖音小店-巨晶建材商行': '594688317@qq.com', '抖音小店-朋意家居生活旗舰店': '594688317@qq.com',
             '抖音小店-舒清好物优选': '594688317@qq.com', '抖音小店-御恒好物优选': '594688317@qq.com',
             '抖音小店-朋意家居旗舰店': '594688317@qq.com', '抖音小店-兴峰建材专营店': '594688317@qq.com',
             '抖音小店-首力金属材料旗舰店': '791002765@qq.com', '抖音小店-朋意箬水贸易专卖店': '791002765@qq.com',
             '抖音小店-汇蕊好物优选': '791002765@qq.com', '抖音小店-首力全屋定制旗舰店': '791002765@qq.com',
             '抖音小店-首力家具旗舰店': '791002765@qq.com',
             '抖音小店-朋意尾鸯家居生活专卖店': '791002765@qq.com', '抖音小店-汇美好物优选': '791002765@qq.com',
             '抖音小店-荟财好物优选': '791002765@qq.com', '抖音小店-天天好价优选': '791002765@qq.com',
             '抖音小店-首力家装旗舰店': 'yexiong1108@qq.com', '抖音小店-朋意厨房旗舰店': '791002765@qq.com',
             '抖音小店-朋意个护保健旗舰店': '791002765@qq.com',
             '抖音小店-朋意楷画家居专卖店': '791002765@qq.com', '抖音小店-美馨好物优选': '791002765@qq.com',
             '抖音小店-朋意澈匀家居日用专卖店': '791002765@qq.com', '抖音小店-旭扬好物优选': '791002765@qq.com',
             '抖音小店-朋意义乌市璟也贸易有限公司家居专卖店': '791002765@qq.com',
             '抖音小店-义乌市昀创贸易有限公司': '791002765@qq.com', '抖音小店-真子收纳用品旗舰店': '791002765@qq.com',
             '抖音小店-家装好物优选': '791002765@qq.com', '抖音小店-义乌市司兔电子商务有限公司': '791002765@qq.com',
             '抖音小店-朋意电器旗舰店': '791002765@qq.com', '抖音小店-朋意厨房电器旗舰店': '791002765@qq.com',
             '抖音小店-斯荣好物优选': '2968425169@qq.com',
             '抖音小店-汇伊好物优选': '2968425169@qq.com', '抖音小店-嘉魅好物优选': '2968425169@qq.com',
             '抖音小店-甜馨好物优选': '2968425169@qq.com', '抖音小店-瑞晶好物优选': '2968425169@qq.com',
             '抖音小店-朋意旅行用品旗舰店': '2968425169@qq.com', '抖音小店-朋意爻允家居专卖店': '2968425169@qq.com',
             '抖音小店-美歆好物优选': '2968425169@qq.com',
             '抖音小店-朋意密洁家居专卖店': '2968425169@qq.com',
             '抖音小店-朋意执情家居专卖店': '2968425169@qq.com', '抖音小店-朋意昀涵家居专卖店': '2968425169@qq.com',
             '抖音小店-朋意家居用品旗舰店': '2968425169@qq.com',
             '抖音小店-首力玄策五金专卖店': '2968425169@qq.com', '抖音小店-朋意嫦昀专卖店': '2968425169@qq.com',
             '抖音小店-首力栩良五金专卖店': '2968425169@qq.com', '抖音小店-朋意展昀生活专卖店': '2968425169@qq.com',
             '抖音小店-朋意辜壹专卖店': '2968425169@qq.com', '抖音小店-盈美好物优选': '2968425169@qq.com',
             '抖音小店-首力影焰专卖店': '2968425169@qq.com', '抖音小店-首力韵霆专卖店': '2968425169@qq.com',
             '抖音小店-朋意益昀家居专卖店': '2968425169@qq.com', '抖音小店-朋意岱宗家居专卖店': 'yexiong1108@qq.com',
             '抖音小店-首力月啼暇专卖店': 'yexiong1108@qq.com', '抖音小店-朋意服饰配件旗舰店': 'yexiong1108@qq.com',
             '抖音小店-朋意厨房用品旗舰店': 'yexiong1108@qq.com',
             '抖音小店-朋意义乌市尥远电子商务有限公司收纳用品专卖店': 'yexiong1108@qq.com',
             '抖音小店-朋意好物优选': 'yexiong1108@qq.com', '抖音小店-朋意贵岚专卖店': 'yexiong1108@qq.com',
             '抖音小店-首力机械旗舰店': 'yexiong1108@qq.com', '抖音小店-首力智能家具旗舰店': 'yexiong1108@qq.com',
             '抖音小店-朋意装饰品旗舰店': 'yexiong1108@qq.com', '抖音小店-朋意尥玥家居专卖店': 'yexiong1108@qq.com',
             '抖音小店-尥玥好物严选': 'yexiong1108@qq.com',
             '抖音小店-首力工具旗舰店': 'yexiong1108@qq.com', '抖音小店-朋意悦昀贸易家居专卖店': 'yexiong1108@qq.com',
             '抖音小店-朋意杭信专卖店': 'yexiong1108@qq.com', '抖音小店-朋意珺驰贸易专卖店': 'yexiong1108@qq.com',
             '抖音小店-朋意仕盷家居专卖店': 'yexiong1108@qq.com', '抖音小店-朋意家纺旗舰店': 'yexiong1108@qq.com',
             '抖音小店-朋意配饰旗舰店': 'yexiong1108@qq.com', '抖音小店-朋意翳昀家居日用专卖店': 'yexiong1108@qq.com',
             '抖音小店-朋意家饰旗舰店': 'yexiong1108@qq.com', '抖音小店-朋意环旭专卖店': 'y2968425169@foxmail.com',
             '抖音小店-朋意昀辜家居生活专卖店': 'y2968425169@foxmail.com',
             '抖音小店-朋意翳个人护理专卖店': 'y2968425169@foxmail.com',
             '抖音小店-首力昀极厨卫五金专卖店': 'y2968425169@foxmail.com',
             '抖音小店-首力首悦贸易家装主材专卖店': 'y2968425169@foxmail.com',
             '抖音小店-首力翳盷家装主材专卖店': 'y2968425169@foxmail.com', }
店铺更名_dict = {'朋意义乌市海辉服饰有限公司': '朋意义乌市海辉服饰有限公司烹饪用具专卖店',
                 '真子家居日用旗舰店': '真子义乌市炳超贸易有限公司家居生活专卖店',
                 '巨晶旗舰店': '巨晶昀沐家居日用专卖店',
                 '首力五金工具旗舰店': '首力义乌市颂伊贸易有限公司五金工具专卖店',
                 '抖in优选供应链': '义乌市烨卦电子商务商行', '淘淘家穿戴甲': '朋意好物优选',
                 '朋意均玥家居专卖店': '朋意电器旗舰店',
                 '首力义乌市首旺电子商务有限公司': '首力义乌市首旺电子商务有限公司建材专卖店',
                 '朋意个护保健旗舰店': '朋意厨房旗舰店',
                 '义乌市枕歌电子商务有限公司': '朋意厨房旗舰店',
                 '朋意厨房电器旗舰店': '朋意楷画家居专卖店',
                 '朋意生活旗舰店': '朋意尾鸯家居生活专卖店',
                 '朋意义乌市尾鸯贸易': '朋意尾鸯家居生活专卖店',
                 '首力家具旗舰店': '首力全屋定制旗舰店', '真子收纳用品旗舰店': '义乌市司兔电子商务有限公司',
                 '朋意电器旗舰店': '朋意箬水贸易专卖店', '真子家居旗舰店': '义乌市昀创贸易有限公司',
                 '朋意家具旗舰店': '朋意澈匀家居日用专卖店',
                 '朋意澈匀电子专卖店': '朋意澈匀家居日用专卖店',
                 '朋意嫦昀专卖店': '朋意家居日用旗舰店', '首力月啼暇专卖店': '首力工具旗舰店',
                 '朋意厨房用品旗舰店': '朋意服饰配件旗舰店',
                 '朋意义乌市尥远电子商务有限公司收纳用品专卖店': '朋意服饰配件旗舰店',
                 '朋意密洁家居专卖店': '朋意厨房用品旗舰店', '朋意佳妙贸易': '朋意昀辜家居生活专卖店',
                 '朋意尥昀家居专卖店': '朋意厨房电器旗舰店', '朋意可悦贸易': '朋意翳昀家居日用品专卖店',
                 '朋意家居用品旗舰店': '朋意义乌市璟也贸易有限公司家居专卖店',
                 '朋意爻允家居专卖店': '朋意旅行用品旗舰店', '朋意昀涵家居专卖店': '朋意家居用品旗舰店',
                 '朋意展昀生活专卖店': '朋意个护用具旗舰店', '尥玥好物严选': '朋意尥玥家居专卖店'}
usernames_list = [usernames1, usernames2, usernames3, usernames4, usernames5, usernames6]
# 获取所需定责的订单列表
shopname_dirtlist = request_get()
print(shopname_dirtlist)
for shopname in shopname_dirtlist.keys():
    print(shopname)
    username = shop_dirt[shopname]
    for usernames in usernames_list:
        # 判断店铺是否在该抖音账号中
        if username in usernames.values():
            # 用该账号登录
            login(usernames)
            port = usernames['port']
            user_path = usernames['user_path']
            downpath = usernames['down_path']
            page = get_page(port, user_path, downpath)
            page.get("https://fxg.jinritemai.com/ffa/multiShop")
            page.wait(5)
            if shopname.replace('抖音小店-', '') in 店铺更名_dict:
                page(店铺更名_dict[shopname.replace('抖音小店-', '')]).click(by_js=True)
            else:
                page(shopname.replace('抖音小店-', '')).click(by_js=True)
            page.wait(5)
            if page.eles('.feige_headerIcon__3CdVk', timeout=5):
                page('.feige_headerIcon__3CdVk').click(by_js=True)
            elif page.s_ele('text=接待'):
                page('text=接待').click(by_js=True)
            page.wait(6)
            # 切换到最新打开的标签页
            tab = page.get_tab(page.latest_tab)
            tab.wait(3)
            if tab.s_eles("退出引导"):
                tab("退出引导").click()
                tab.wait(1)
            if tab.s_eles('.qg8dJkHI2UbLch1KK8J6'):
                tab('.qg8dJkHI2UbLch1KK8J6').ele("tag:img").click()
                tab.wait(1)
            # 遍历订单列表并逐条处理
            for order in shopname_dirtlist[shopname]:
                order_id = order['orderNo']
                order_shopname = order['shopName']
                task_id = order['taskId']
                # 输入查询的订单号
                tab('@placeholder=用户昵称/抖音号/订单号(Ctrl+F)').input(order_id, clear=True)
                tab.wait(1.5)
                # 打开对应订单聊天窗口
                try:
                    tab('#chantListScrollArea').child(2).click.multi(2)
                except:
                    response_data(task_id, 2, [], [])
                    continue
                tab.wait(3)
                # 获取聊天数据
                chatRecordList, fileUrls = get_message_data(tab)
                response_data(task_id, 1, fileUrls, chatRecordList)
                continue
            tab.close()
            break
